使用python构建卷积神经网络实现猫狗的分类问题

最新推荐文章于 2024-06-18 20:00:32 发布

wopoli

最新推荐文章于 2024-06-18 20:00:32 发布

阅读量2.3k

点赞数 1

分类专栏： python 文章标签：神经网络 python pytorch

本文链接：https://blog.csdn.net/wopoli/article/details/120085491

版权

本文介绍了如何使用Python和PyTorch构建卷积神经网络来解决猫狗分类问题。从导入数据集到构建并训练模型，详细阐述了每个步骤，最终展示训练结果。

摘要由CSDN通过智能技术生成

使用python构建卷积神经网络实现猫狗的分类问题

需求：使用卷积神经网络实现猫狗的二分类问题。
使用的语言：python
使用的工具：pytorch、pyCharm
数据集：链接：https://pan.baidu.com/s/1zfPFfmAjQ6ERJSQ4zEQHbw
提取码：vkk5

1、导入数据集

# 导入包
import os		# 用于当前文件
# 用于构建神经网络
import torch		
from torch import nn, optim	
from torch.nn import functional as F
from torch.utils import data
from torchvision import datasets
from torchvision import transforms

ImageSize = 224     # 定义图片大小
train_dir = './train/'    # 定义训练集存储路径
test_dir = './test/'    # 定义测试集存储路径
# 定义数据集预处理方法
data_transform = transforms.Compose([
    transforms.Resize((ImageSize, ImageSize)),    # 重新调整图片大小
    transforms.ToTensor(),       # 将图片转换为Tensor数组
    transforms.Normalize((0.1307,), (0.3081,))		# 归一化操作
])

# datasets.ImageFolder(dir, transform)：导入数据集
# dir：数据集存储路径
# transform：数据集预处理方法
datasets_train = dataset